how to process complex json files in pyspark